
import torch
print(torch.__version__)
torch.cuda.is_available()
torch.cuda.device_count()
import torchvision
print(torchvision.__version__)
print(torch.cuda.is_available())
quit()